clu
jax
flax
ml-collections
numpy
tensorflow
tensorflow-datasets
tensorflow-privacy